from otree.api import ( models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer, Currency as c, currency_range, ) import random import csv import numpy doc = """ This is a one-shot "Prisoner's Dilemma". Two players are asked separately whether they want to cooperate or defect. Their choices directly determine the payoffs. """ SUPERGROUP_NUM_ERR = 'Wrong number of players per supergroup' class Constants(BaseConstants): name_in_url = 'strategic_draft1b' players_per_group = 2 players_per_supergroup = 4 assert players_per_supergroup % players_per_group == 0, \ SUPERGROUP_NUM_ERR # num_rounds = 2 num_rounds = 5 supergroup_threshold1=4 instructions_template = 'strategic_draft0/instructions.html' # payoff if 1 player defects and the other cooperates""", betray_payoff = c(100) betrayed_payoff = c(10) # payoff if both players cooperate or both defect both_cooperate_payoff = c(73) both_defect_payoff = c(43) def slice_list(input): ppg = Constants.players_per_group output = [input[i:i+ppg] for i in range(0, len(input), ppg)] for o in output: assert len(o) == ppg, SUPERGROUP_NUM_ERR return output class Subsession(BaseSubsession): def creating_session(self): if self.round_number == 1: assert len(self.get_players()) % \ Constants.players_per_supergroup == 0, \ SUPERGROUP_NUM_ERR import itertools random_list = [1, 2, 3, 4, 5, 6, 7, 8] random.shuffle(random_list) random_ids = itertools.cycle(random_list) for p in self.get_players(): p.random_id = next(random_ids) for g in self.get_groups(): for p in g.get_players(): if p.random_id <= Constants.supergroup_threshold1: p.participant.vars['supergroup'] = int(1) else: p.participant.vars['supergroup'] = int(2) p.participant.vars['id_random'] = int(p.random_id) subgroups = set([p.vars['supergroup'] for p in self.session.get_participants()]) new_matrix = [] for s in subgroups: A = [p for p in self.get_players() if p.participant.vars['supergroup'] == s] random.shuffle(A) sliced_supergroup = slice_list(A) new_matrix.extend(sliced_supergroup) print(new_matrix) self.set_group_matrix(new_matrix) with open('strategic_draft0/matching_matrix.csv', 'w') as csvfile: writer = csv.writer(csvfile) writer.writerows(new_matrix) with open('strategic_draft0/matching_matrix2.csv', 'w+') as my_csv: csvWriter = csv.writer(my_csv, delimiter=',') csvWriter.writerows(new_matrix) #numpy.savetxt('strategic_draft0/matching_matrix3.csv', new_matrix, delimiter=",") #if self.round_number == 1: # assign a random number to participants [1,N] #for p in self.get_players(): #p.participant.vars['id_random'] = random.choice([1,2,3,4,5,6,7,8]) class Group(BaseGroup): def set_payoffs(self): for p in self.get_players(): p.set_payoff() # for p in self.get_players(): # if p.round_number==1: # p.cooprate_r1 = self.total_coop # p.totalcoop_r1 = self.coop_rate # self.total_coop = sum([p.decision for p in self.get_players_by_role('1')]) # self.coop_rate = int(self.total_coop / 8 * 100) # def set_cooprates(self): # #s1 = [p for p in self.get_players() if p.role() == '1'] # #s2 = [p for p in self.get_players() if p.role() == '2'] # #s1 = self.get_player_by_role('1') # #s2 = self.get_player_by_role('2') # # self.total_coop = sum([p.decision for p in s1]) # # self.total_coop = sum([p.decision for p in s2]) # # self.total_coop_s1 = sum([p.decision for p in s1]) # # self.total_coop_s2 = sum([p.decision for p in s2]) # for g in self.get_groups(): # for p in g.get_players(): # if p.role() == '1': # p.total_coop_s1_temp += self.decision # p.num_s1_players += 1 # p.num_s1_players += 0 # #self.total_coop_s2_temp = 0 # #self.num_s2_players += 0 # if p.role() == '2': # #self.total_coop_s1_temp += None # #self.num_s1_players += None # p.total_coop_s2_temp += player.decision # p.num_s2_players += 1 class Player(BasePlayer): decision = models.IntegerField( choices=[[1 , 'Cooperate'], [0 , 'Defect']], doc="""This player's decision""", widget=widgets.RadioSelect, ) decision_guess= models.IntegerField(choices=[[1 , '0%-20%'], [2 , '21%-40%'], [3 , '41%-60%'], [4, '61%-80%'],[5,'81%-100%']], min=1, max=10, widget=widgets.RadioSelectHorizontal, ) #decision_guess = models.IntegerField(choices=[[1, '0%-10%'], [2, '11%-20%'], [3, '21%-30%'], # [4, '31%-40%'], [5, '41%-50%'], [5, '41%-50%'], # [6, '51%-60%'],[7, '61%-70%'], [8, '71%-80%'], [9, '81%-90%'], [10, '91%-100%']], # min=1, max=10, # widget=widgets.RadioSelectHorizontal, #) supergroup = models.LongStringField() random_id = models.IntegerField() decision_type = models.StringField() decision_r1 = models.IntegerField() decision_r2 = models.IntegerField() decision_r3 = models.IntegerField() decision_r4 = models.IntegerField() decision_r5 = models.IntegerField() decision_type_r1 = models.StringField() decision_type_r2 = models.StringField() decision_type_r3 = models.StringField() decision_type_r4 = models.StringField() decision_type_r5 = models.StringField() decision_guess_r1 = models.LongStringField() decision_guess_r5 = models.LongStringField() payoff_r1 = models.CurrencyField() payoff_r2 = models.CurrencyField() payoff_r3 = models.CurrencyField() payoff_r4 = models.CurrencyField() payoff_r5 = models.CurrencyField() #payoff_r1_int = models.IntegerField() #payoff_r2_int = models.IntegerField() payoff_int = models.IntegerField() cooprate_r1 = models.IntegerField() cooprate_r2 = models.IntegerField() cooprate_r3 = models.IntegerField() cooprate_r4 = models.IntegerField() cooprate_r5 = models.IntegerField() #totalcoop_r1 = models.IntegerField() total_coop = models.IntegerField() total_coop_s1 = models.IntegerField() total_coop_s2 = models.IntegerField() coop_rate_s1 = models.IntegerField() coop_rate_s2 = models.IntegerField() #num_s1_players = models.IntegerField() #num_s2_players = models.IntegerField() def other_player(self): return self.get_others_in_group()[0] def set_payoff(self): payoff_matrix = dict( Cooperate=dict( Cooperate=Constants.both_cooperate_payoff, Defect=Constants.betrayed_payoff, ), Defect=dict( Cooperate=Constants.betray_payoff, Defect=Constants.both_defect_payoff ), ) self.payoff = payoff_matrix[self.decision_type][self.other_player().decision_type] self.supergroup = str(self.participant.vars['supergroup']) self.random_id = int(self.participant.vars['id_random']) if self.decision == 1 and self.other_player().decision == 1: self.payoff_int = int(Constants.both_cooperate_payoff) elif self.decision == 0 and self.other_player().decision == 0: self.payoff_int = int(Constants.both_defect_payoff) elif self.decision == 1 and self.other_player().decision == 0: self.payoff_int = int(Constants.betrayed_payoff) else: self.payoff_int = int(Constants.betray_payoff) if self.round_number == 1: self.payoff_r1 = self.in_round(self.round_number).payoff # self.cooprate_r1 = sum(self.decision)/Constants.players_per_supergroup*100 # else: # self.cooprate_r1 = sum(self.decision)/Constants.players_per_supergroup*100 if self.round_number == 2: self.payoff_r2 = self.in_round(self.round_number).payoff if self.round_number == 3: self.payoff_r3 = self.in_round(self.round_number).payoff if self.round_number == 4: self.payoff_r4 = self.in_round(self.round_number).payoff if self.round_number == 5: self.payoff_r5 = self.in_round(self.round_number).payoff def role(self): if self.participant.vars['supergroup'] == int(1): return '1' else: return '2'